# -*- coding: utf-8 -*-
"""
Created on Mon Aug 26 13:36:59 2019

@author: ikaros
"""

"""
jpg图片和png图片的区别
jpg格式:是有损图片压缩类型,可用最少的磁盘空间得到较好的图像质量
png格式:不是压缩性,能保存透明等图

"""
from PIL import Image
import cv2
import os
import numpy as np


def PNG_JPG(PngPath):
    img = cv2.imread(PngPath, 0)
    w, h = img.shape[::-1]
    infile = PngPath
    outfile = os.path.splitext(infile)[0] + ".jpg"
    img = Image.open(infile)
    img = img.resize((int(w / 2), int(h / 2)), Image.ANTIALIAS)
    try:
        if len(img.split()) == 4:
            # prevent IOError: cannot write mode RGBA as BMP
            r, g, b, a = img.split()
            img = Image.merge("RGB", (r, g, b))
            img.convert('RGB').save(outfile, quality=70)
            os.remove(PngPath)
        else:
            img.convert('RGB').save(outfile, quality=70)
            os.remove(PngPath)
        return outfile
    except Exception as e:
        print("PNG转换JPG 错误", e)

files_dir = "test1/"
for root,dirs,files in os.walk(files_dir):
    print(files)
    
files2 = np.array(files)
print(files2.shape)

row1 = files2.shape[0]
for i in range(row1):
    str1 = files2[i]
    PngPath = files_dir + str1
    print(PngPath)
    img = cv2.imread(PngPath)
    
    infile = PngPath
    outfile = os.path.splitext(infile)[0] + ".jpg"
    print(outfile)
    cv2.imwrite(outfile,img)
    
    
